Add DGX Spark / GB10 fork to notable forks#207
Open
laudney wants to merge 1 commit intokarpathy:masterfrom
Open
Add DGX Spark / GB10 fork to notable forks#207laudney wants to merge 1 commit intokarpathy:masterfrom
laudney wants to merge 1 commit intokarpathy:masterfrom
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Adds our fork for the NVIDIA DGX Spark (GB10 chip, Blackwell sm_121a, 128GB unified memory).
The upstream code targets H100 and uses FA3 kernels +
kernelspackage, neither of which work on the GB10's sm_121a architecture. This fork gets autoresearch running out-of-the-box on DGX Spark with the following changes:flex_attention— FA3 kernels are Hopper-only (sm_90). flex_attention uses Triton-generated kernels that work on any architecture including Blackwell sm_121a, with sliding window support viacreate_block_maskkernelsdependency — not needed once FA3 is replaced with flex_attentionWe also benchmarked FA4 (blake-snc sm120 fork) and cuDNN SDPA as alternative attention backends — both showed parity or worse vs flex_attention on this hardware. The GB10 is memory-bandwidth limited at 273 GB/s, so most kernel-level optimizations don't move the needle.